/*
 * Copyright (c) 2024 Huawei Device Co., Ltd.
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#ifndef ARK_WEB_ENGINE_WRAPPER_H_
#define ARK_WEB_ENGINE_WRAPPER_H_
#pragma once

#include "include/nweb_engine.h"
#include "ohos_nweb/include/ark_web_engine.h"

namespace OHOS::ArkWeb {

class ArkWebEngineWrapper : public OHOS::NWeb::NWebEngine {
public:
  ArkWebEngineWrapper(ArkWebRefPtr<ArkWebEngine> ark_web_engine);
  ~ArkWebEngineWrapper() = default;

  std::shared_ptr<OHOS::NWeb::NWeb>
  CreateNWeb(std::shared_ptr<OHOS::NWeb::NWebCreateInfo> create_info) override;

  std::shared_ptr<OHOS::NWeb::NWeb> GetNWeb(int32_t nweb_id) override;

  std::shared_ptr<OHOS::NWeb::NWebDataBase> GetDataBase() override;

  std::shared_ptr<OHOS::NWeb::NWebWebStorage> GetWebStorage() override;

  std::shared_ptr<OHOS::NWeb::NWebCookieManager> GetCookieManager() override;

  std::shared_ptr<OHOS::NWeb::NWebDownloadManager>
  GetDownloadManager() override;

  void SetWebTag(int32_t nweb_id, const char *web_tag) override;

  void InitializeWebEngine(
      std::shared_ptr<OHOS::NWeb::NWebEngineInitArgs> init_args) override;

  void PrepareForPageLoad(const std::string &url, bool preconnectable,
                          int32_t num_sockets) override;

  void SetWebDebuggingAccess(bool isEnableDebug) override;

  void AddIntelligentTrackingPreventionBypassingList(
      const std::vector<std::string> &hosts) override;

  void RemoveIntelligentTrackingPreventionBypassingList(
      const std::vector<std::string> &hosts) override;

  void ClearIntelligentTrackingPreventionBypassingList() override;

  void PauseAllTimers() override;

  void ResumeAllTimers() override;

  void PrefetchResource(const std::shared_ptr<OHOS::NWeb::NWebEnginePrefetchArgs> &pre_args,
                        const std::map<std::string, std::string> &additional_http_headers,
                        const std::string &cache_key,
                        const uint32_t &cache_valid_time) override;

private:
  ArkWebRefPtr<ArkWebEngine> ark_web_engine_;
};

} // namespace OHOS::ArkWeb

#endif // ARK_WEB_ENGINE_WRAPPER_H_
